import os
from shutil import copyfile
from generate_license_plate import LicensePlateGenerator


if __name__ == '__main__':

    plate_num_str_list = []
    i = 1
    for root, dirs, files in os.walk("D:\\temp\\gan_chepai4\\testB"):
        for name in files:
            plate_num_str = name[:7]
            plate_num_str_list.append(plate_num_str)
            copyfile("D:\\temp\\gan_chepai4\\testB\\" + name, "D:\\temp\\gan_chepai4\\testC\\" + str(i) + "_" + name)
            i += 1

    LicensePlateGenerator.generate_license_plate_images('single_blue', plate_num_str_list, "D:\\temp\\gan_chepai4\\testA")

